package com.whty.cms.filewps.util;

import com.whty.cms.filewps.common.constants.Consts;

import java.util.regex.Pattern;

/**
 * 文件格式判断工具类
 * Created by Administrator on 2019/3/6 0006.
 */
public enum  FileFormat {
    FILE_FORMAT_DOCUMENT("doc"),        //文档类格式
    FILE_FORMAT_VIDEO("video"),         //视频类格式
    FILE_FORMAT_AUDIO("audio"),         //音频类格式
    FULE_FORMAT_PICTURE("pic"),         //图片类格式
    FULE_FORMAT_OTHER("other");       //其他类型

    String getFileType;

    FileFormat(String type) {
        this.getFileType = type;
    }
    public String getGetFileType() {
        return getFileType;
    }

    /**
     * 根据文件名称返回文件格式类型
     * @param fileName
     * @return
     */
    public static String getFileType(String fileName) {
        if(Pattern.compile(Consts.FILE_PIC).matcher(fileName).matches()) {
            return FileFormat.FULE_FORMAT_PICTURE.getFileType;
        } else if (Pattern.compile(Consts.FILE_TRANS_COS_AUDIO).matcher(fileName).matches()) {
            return FileFormat.FILE_FORMAT_VIDEO.getFileType;
        } else if (Pattern.compile(Consts.FILE_TRANS_COS_AUDIO).matcher(fileName).matches()) {
            return FileFormat.FILE_FORMAT_AUDIO.getFileType;
        } else if (Pattern.compile(Consts.FILE_TRANS_DOCUMENT).matcher(fileName).matches()) {
            return FileFormat.FILE_FORMAT_DOCUMENT.getFileType;
        } else {
            return FileFormat.FULE_FORMAT_OTHER.getFileType;
        }
    }
}
